import os
from dotenv import load_dotenv
from processors.video_processor import process_chapter
from static_configuration import contents

# 加载环境变量
load_dotenv()


def main():
    # 从环境变量获取API密钥
    api_key = os.environ.get("ARK_API_KEY")
    if not api_key:
        print("❌ 错误: 未设置 ARK_API_KEY 环境变量")
        return

    chapter_content = contents
    result = process_chapter(chapter_content, api_key)

    print("\n" + "=" * 60)
    print("🎬 视频生成完成!")
    print("=" * 60)

    if result["success"]:
        print(f"✅ 成功生成视频")
        print(f"📁 输出文件夹: {result['output_dir']}")
        print(f"🕐 生成时间戳: {result['timestamp']}")
        print(f"📽️ 最终视频: {result.get('output_file', '单段视频')}")
        print(f"🎞️ 视频段数: {result['segments_count']}")
        print(f"📝 段落数: {result['paragraphs_count']}")
    else:
        print(f"❌ 生成失败: {result['error']}")
        print(f"📁 输出文件夹: {result.get('output_dir', '未创建')}")

    print("=" * 60)


if __name__ == "__main__":
    main()
